package com.zx.ecode;

import com.zx.ecode.model.entity.EpidemicBean;
import com.zx.ecode.model.entity.TripBean;
import com.zx.ecode.model.entity.UserBean;

import org.json.JSONArray;
import org.json.JSONException;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ThreadLocalRandom;

public class DataExampleUtil {

    public static UserBean getUser_1(int user) {
        UserBean userBean = new UserBean();
        switch (user) {
            case 1:
                userBean.setAge("18");
                userBean.setFaceImg("");
                userBean.setId("1");
                userBean.setId_number("41351961256126635");
                userBean.setName("张三");
                userBean.setRisk_level(1);
                userBean.setTrip(getTrip(1));
                userBean.setVaccine_status(0);
                break;
            case 2:
                userBean.setAge("26");
                userBean.setFaceImg("");
                userBean.setId("2");
                userBean.setId_number("56384512359874236");
                userBean.setName("李四");
                userBean.setRisk_level(2);
                userBean.setTrip(getTrip(2));
                userBean.setVaccine_status(1);
                break;
            case 3:
                userBean.setAge("33");
                userBean.setFaceImg("");
                userBean.setId("3");
                userBean.setId_number("546381254698752135");
                userBean.setName("王五五");
                userBean.setRisk_level(3);
                userBean.setTrip(getTrip(3));
                userBean.setVaccine_status(2);
                break;
            case 4:
                userBean.setAge("23");
                userBean.setFaceImg("");
                userBean.setId("4");
                userBean.setId_number("146587132549032568");
                userBean.setName("赵老六");
                userBean.setRisk_level(3);
                userBean.setTrip(getTrip(4));
                userBean.setVaccine_status(3);
                break;
            case 5:
                userBean.setAge("21");
                userBean.setFaceImg("");
                userBean.setId("5");
                userBean.setId_number("031204578193023568");
                userBean.setName("田七");
                userBean.setRisk_level(1);
                userBean.setTrip(getTrip(5));
                userBean.setVaccine_status(0);
                break;
        }
        return userBean;
    }


    private static List<TripBean> getTrip(int level) {
        List<TripBean> list = new ArrayList<>();
        List<TripBean> trip1 = new ArrayList<>();
        List<TripBean> trip2 = new ArrayList<>();
        List<TripBean> trip3 = new ArrayList<>();
        List<TripBean> trip4 = new ArrayList<>();
        List<TripBean> trip5 = new ArrayList<>();
        TripBean tripBean1 = new TripBean();
        tripBean1.setLevel(1);
        tripBean1.setName("南京");
        tripBean1.setTime("2021-08-12");
        list.add(tripBean1);

        TripBean tripBean2 = new TripBean();
        tripBean2.setLevel(1);
        tripBean2.setName("郑州");
        tripBean2.setTime("2021-08-19");
        list.add(tripBean2);

        TripBean tripBean3 = new TripBean();
        tripBean3.setLevel(2);
        tripBean3.setName("哈尔滨");
        tripBean3.setTime("2021-08-01");
        list.add(tripBean3);

        TripBean tripBean4 = new TripBean();
        tripBean4.setLevel(1);
        tripBean4.setName("深圳");
        tripBean4.setTime("2021-09-03");
        list.add(tripBean4);

        TripBean tripBean5 = new TripBean();
        tripBean5.setLevel(3);
        tripBean5.setName("荥阳");
        tripBean5.setTime("2021-10-19");
        list.add(tripBean5);

        TripBean tripBean6 = new TripBean();
        tripBean6.setLevel(1);
        tripBean6.setName("北京");
        tripBean6.setTime("2021-11-01");
        list.add(tripBean6);

        TripBean tripBean7 = new TripBean();
        tripBean7.setLevel(1);
        tripBean7.setName("上海");
        tripBean7.setTime("2021-11-10");
        list.add(tripBean7);

        TripBean tripBean8 = new TripBean();
        tripBean8.setLevel(1);
        tripBean8.setName("南阳");
        tripBean8.setTime("2021-10-21");
        list.add(tripBean8);

        TripBean tripBean9 = new TripBean();
        tripBean9.setLevel(2);
        tripBean9.setName("福建");
        tripBean9.setTime("2021-10-13");
        list.add(tripBean9);

        TripBean tripBean10 = new TripBean();
        tripBean10.setLevel(1);
        tripBean10.setName("呼和浩特");
        tripBean10.setTime("2021-10-07");
        list.add(tripBean10);

        TripBean tripBean11 = new TripBean();
        tripBean11.setLevel(1);
        tripBean11.setName("泉州");
        tripBean11.setTime("2021-11-05");
        list.add(tripBean11);

        trip1.add(tripBean1);
        trip1.add(tripBean2);
        trip1.add(tripBean3);
        trip1.add(tripBean4);
        trip1.add(tripBean6);
        trip1.add(tripBean7);

        trip2.add(tripBean4);
        trip2.add(tripBean2);
        trip2.add(tripBean7);
        trip2.add(tripBean11);
        trip2.add(tripBean10);

        trip3.add(tripBean8);
        trip3.add(tripBean7);
        trip3.add(tripBean6);
        trip3.add(tripBean4);
        trip3.add(tripBean2);
        trip3.add(tripBean10);
        trip3.add(tripBean1);

        trip4.add(tripBean2);
        trip4.add(tripBean1);
        trip4.add(tripBean5);
        trip4.add(tripBean3);
        trip4.add(tripBean10);
        trip4.add(tripBean8);
        trip4.add(tripBean4);

        trip5.add(tripBean11);
        trip5.add(tripBean9);
        trip5.add(tripBean10);
        trip5.add(tripBean2);
        trip5.add(tripBean7);
        trip5.add(tripBean6);
        trip5.add(tripBean4);
        switch (level) {
            case 1:
                return trip2;
            case 2:
                return trip5;
            case 3:
                return trip4;
            case 4:
                return trip1;
            case 5:
                return trip3;
        }
        return list;
    }

    public static List<EpidemicBean> getEpidemic() throws JSONException {
        String str = "[{\"title\":\"新冠口服药问世 能否终结全球疫情?\",\"content\":\"近期，新冠口服药研发领域捷报频传。11月4日，全球首个新冠口服药获批，英国药品和保健品管理局宣布批准默沙东的莫努匹韦（Molnupiravir）上市；11月5日，辉瑞也公布其新冠小分子口服药Paxlovid的2\\\\3期双盲对照临床试验结果，数据显示，Paxlovid可使住院或死亡风险降低89%。\"},{\"title\":\"本轮疫情已波及21个省区市!\",\"content\":\"据国家卫健委网站今天通报，昨天0至24时，31个省区市和新疆生产建设兵团报告新增确诊病例54例。其中境外输入病例15例，本土病例39例；截至昨天24时，据31个省区市和新疆生产建设兵团报告，现有确诊病例1222例（其中重症病例21例），新增无症状感染者39例。\"},{\"title\":\"直击国门一线“战疫”坚守者:不胜疫情不收兵\",\"content\":\"国门口岸是严防境外疫情输入国内的重要关口。连日来,国内外疫情呈现多点散发趋势,重庆出入境边防检查总站实施封闭执勤、集中隔离、居家监测的勤务模式,将相关人员安排在机场周边集中居住,通过高防疫标准执勤、专车接送保障、餐食单独配送等\"},{\"title\":\"疫情中,科技为我们做了什么?\",\"content\":\"伴随来势汹汹的寒潮，新冠疫情也在近期席卷了十余省市，累积报告病例达数百人。低风险地区开始逐个排查国庆节前后流动人员的行程及健康状况，中高风险地区又开始了全民核酸。不管你是沿海还是内陆，不管你是一线还是十三线城市，新冠病例的出现总是来的让人措手不及。谁能想到，去迪士尼玩不光进门要排长队，出门还得排核酸检测的长队呢？\"},{\"title\":\"国家卫健委权威发布，本轮疫情源头已查清！\",\"content\":\"国家卫健委疾控局副局长吴良有表示，自10月17日以来，至11月5日24时，全国累计报告感染者918例，此轮疫情波及20省份44市。病毒基因测序和流调溯源结果显示，本轮疫情由多个不关联的境外输入源头引起。总体上，本次疫情呈现传播链条多、传播速度快、传播范围广的特点，外防输入、内防反弹的防控形势依然严峻复杂。\"}]";
        List<EpidemicBean> list = new ArrayList<>();
        EpidemicBean epidemicBean = new EpidemicBean();
        JSONArray array = new JSONArray(str);
        for (int i = 0; i < array.length(); i++) {
            epidemicBean.setTitle(array.getJSONObject(i).getString("title"));
            epidemicBean.setContent(array.getJSONObject(i).getString("content"));
            list.add(epidemicBean);
        }
        return list;
    }
}